package com.mail;

import com.rabbitmq.client.Channel;
import com.students.pojo.Employee;
import com.students.pojo.MailConstants;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.amqp.support.AmqpHeaders;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.mail.MailProperties;
import org.springframework.data.redis.core.HashOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.mail.javamail.MimeMessageHelper;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageHeaders;
import org.springframework.stereotype.Component;
import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;

import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import java.io.IOException;
import java.util.Date;

/*
* 邮件信息接受者
* */
@Component
public class MailReceiver {

    private static final Logger LOGGER= LoggerFactory.getLogger(MailReceiver.class);
    @Autowired
    private RedisTemplate redisTemplate;
    @Autowired
    private JavaMailSender javaMailSender;
    @Autowired
    private MailProperties mailProperties;
    @Autowired
    private TemplateEngine templateEngine;

//    @RabbitListener(queues = "mail.welcome")
    @RabbitListener(queues = MailConstants.MAIL_QUEUE_NAME)
    public void handler(Message message, Channel channel){
        Employee payload = (Employee) message.getPayload();
        MessageHeaders headers=message.getHeaders();
        //消息序号
        long tag=(long) headers.get(AmqpHeaders.DELIVERY_TAG);
        System.out.println("tag = " + tag);
        String msgId = (String) headers.get("spring_returned_message_correlation");
        System.out.println("msgId = " + msgId);
        HashOperations hash = redisTemplate.opsForHash();

        try {
            if (hash.entries("mail_log").containsKey(msgId)){
                LOGGER.error("消息已经被消费============{}",msgId);
                channel.basicAck(tag,false);
                return;
            }
            MimeMessage msg=javaMailSender.createMimeMessage();
            MimeMessageHelper helper=new MimeMessageHelper(msg);

            //发件人
            helper.setFrom(mailProperties.getUsername());
            //收件人
            helper.setTo(payload.getEmail());
            //主题
            helper.setSubject("入职邮件");
            //发送日期
            helper.setSentDate(new Date());
            //邮件内容
            Context context=new Context();
            context.setVariable("name",payload.getName());
            context.setVariable("postName",payload.getPosition().getName());
            context.setVariable("joblevelName",payload.getJoblevel().getName());
            context.setVariable("departmentName",payload.getDepartment().getName());

            String mail=templateEngine.process("mail",context);
            helper.setText(mail,true);
            //发送
            javaMailSender.send(msg);
            LOGGER.info("邮件发送成功");
            //将消息id存入redis
            hash.put("mail_log", msgId, "OK");
            System.out.println("MailReceiver: redis---> msgId = " + msgId);
            //手动确认消息
            channel.basicAck(tag, false);
        } catch (Exception e) {
            try {
                channel.basicNack(tag,false,true);
            } catch (IOException ioException) {
                ioException.printStackTrace();
            }
            LOGGER.error("邮件发送失败>========>{}",e.getMessage());
        }
    }
}
